import time

from wordpress_project.base.base_page import Tools


class reg_tool(Tools):
    login_url = 'http://192.168.0.125:9009/wp-login.php?action=register'
    username = '[name="user_login"]'  # 用户名css定位
    emil = '[name="user_email"]'  # 邮箱css定位
    reg_button = '[name="wp-submit"]'  # 注册按钮css定位
    login_success_ele = '[class="message"]'  # 注册成功 之后的断言定位
    login_fail_ele = '[id="login_error"]'  # 注册失败 之后的 断言定位

    def login_action(self, usr, emil):
        # 在方法中调用同一个类的其他方法， self.方法名  在方法中，调用类的变量self.变量名
        self.visit(self.login_url)
        self.by_css(self.username).send_keys(usr)
        time.sleep(1)
        self.by_css(self.emil).send_keys(emil)
        time.sleep(1)
        self.by_css(self.reg_button).click()
        time.sleep(1)
        self.img()  # 调用截图

    def get_success_msg(self):
        return self.by_css(self.login_success_ele).text

    def get_fail_msg(self):
        return self.by_css(self.login_fail_ele).text


# from selenium import webdriver

# d = webdriver.Chrome()
# a = reg_tool(driver=d)
# a.login_action(usr='hfhfaj112', emil='hfhfaj112@qq.com')
# a.quit()
